YOLO格式的DOTA遥感数据集和YoloV5代码 |
您所在的位置：网站首页 › dota 小雨 › YOLO格式的DOTA遥感数据集和YoloV5代码 |
YOLO格式的DOTA遥感数据集和YoloV5代码
|
DOTA_split数据集下载(数据集中的标签是水平框，没有旋转框）一、数据集下载 最近为了训练yolo4和5，下载了DOTA数据集来用，不过下载的数据需要处理后让它符合yolo的标注格式，折腾了一天终于搞定了数据集的问题，现在分享一下数据集的地址（我只处理了V1.0版的DOTA，V1.5的虽然也下载下来了，但是只把其中的png转换成了jpg，其他的没动）： 百度云：https://pan.baidu.com/s/1UX7oX3_x5CrP_SxSA7XKXQ 提取码: iw3w （如果下载速度太慢，B度W盘-会员可以买一天用一下，https://query.xiqqingchun.com/list.html） 里面有两个压缩包，其中DOTA_original是从DOTA官网下载的，DOTA_split是根据yolo要求处理过的； images/ 中的图片已经改为.jpg格式，且已裁剪，因为我在colab训练yolov5的时候用png会报错lables/ 中标注文件已经处理为yolo格式  另附DOTA1.0、 DOTA1.5、 UCAS-AOD、 HRSC2016原始数据集 https://pan.baidu.com/s/1xjXV9MnvV2mzb_rhmhFG-g 密码：jd5y  二、训练代码下载本文使用的Yolov5代码 主要环境：python 3.7 pytorch 1.7.1, torchvision 0.8.1环境安装 ： pip install -r requirements.txt  链接：https://pan.baidu.com/s/1umW9M-X6cWGmLODa2-pzug 提取码：1234  yolo5中dota_data/dota_name.yaml文件中设置的类别： # PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/ # Download command: bash ./data/get_voc.sh # Train command: python train.py --data voc.yaml # Default dataset location is next to /yolov5: # /parent_folder # /VOC # /yolov5 # train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/] train: dota_data/images/train/ # 图片和label路径 val: dota_data/images/val/ # number of classes nc: 15 # class names names: ['small-vehicle', 'large-vehicle', 'plane', 'storage-tank', 'ship', 'harbor', 'ground-track-field','soccer-ball-field', 'tennis-court', 'swimming-pool', 'baseball-diamond', 'roundabout', 'basketball-court', 'bridge', 'helicopter']其中dota_data/yolov5l_dota.yaml是yolov5l模型，可以根据自己的需求换成其他模型，比如s、m等  dota_data/yolov5l_dota.yaml关于yolov5的训练代码使用了2500张train，800 val，size=640*640，训练了300epoch    inference三、某些不要脸的玩意 某鱼上盗取他人成果的废物 直接拿我的训练成果来卖，图片还是我的原图 我的训练结果 
|
【本文地址】
今日新闻 |
推荐新闻 |